home *** CD-ROM | disk | FTP | other *** search
/ AmigActive 25 / AACD 25.iso / AACD / Magazine / Online / QMail / source / env.c < prev    next >
Encoding:
C/C++ Source or Header  |  1997-04-15  |  2.2 KB  |  114 lines
  1. /* env.c, envread.c, env.h: environ library
  2. Daniel J. Bernstein, djb@silverton.berkeley.edu.
  3. Depends on str.h, alloc.h.
  4. Requires environ.
  5. 19960113: rewrite. warning: interface is different.
  6. No known patent problems.
  7. */
  8.  
  9. #include "str.h"
  10. #include "alloc.h"
  11. #include "env.h"
  12.  
  13. int env_isinit = 0; /* if env_isinit: */
  14. static int ea; /* environ is a pointer to ea+1 char*'s. */
  15. static int en; /* the first en of those are ALLOCATED. environ[en] is 0. */
  16.  
  17. static void env_goodbye(i) int i;
  18. {
  19.  alloc_free(environ[i]);
  20.  environ[i] = environ[--en];
  21.  environ[en] = 0;
  22. }
  23.  
  24. static char *null = 0;
  25.  
  26. void env_clear()
  27. {
  28.  if (env_isinit) while (en) env_goodbye(0);
  29.  else environ = &null;
  30. }
  31.  
  32. static void env_unsetlen(s,len) char *s; int len;
  33. {
  34.  int i;
  35.  for (i = en - 1;i >= 0;--i)
  36.    if (!str_diffn(s,environ[i],len))
  37.      if (environ[i][len] == '=')
  38.        env_goodbye(i);
  39. }
  40.  
  41. int env_unset(s) char *s;
  42. {
  43.  if (!env_isinit) if (!env_init()) return 0;
  44.  env_unsetlen(s,str_len(s));
  45.  return 1;
  46. }
  47.  
  48. static int env_add(s) char *s;
  49. {
  50.  char *t;
  51.  t = env_findeq(s);
  52.  if (t) env_unsetlen(s,t - s);
  53.  if (en == ea)
  54.   {
  55.    ea += 30;
  56.    if (!alloc_re(&environ,(en + 1) * sizeof(char *),(ea + 1) * sizeof(char *)))
  57.     { ea = en; return 0; }
  58.   }
  59.  environ[en++] = s;
  60.  environ[en] = 0;
  61.  return 1;
  62. }
  63.  
  64. int env_put(s) char *s;
  65. {
  66.  char *u;
  67.  if (!env_isinit) if (!env_init()) return 0;
  68.  u = alloc(str_len(s) + 1);
  69.  if (!u) return 0;
  70.  str_copy(u,s);
  71.  if (!env_add(u)) { alloc_free(u); return 0; }
  72.  return 1;
  73. }
  74.  
  75. int env_put2(s,t) char *s; char *t;
  76. {
  77.  char *u;
  78.  int slen;
  79.  if (!env_isinit) if (!env_init()) return 0;
  80.  slen = str_len(s);
  81.  u = alloc(slen + str_len(t) + 2);
  82.  if (!u) return 0;
  83.  str_copy(u,s);
  84.  u[slen] = '=';
  85.  str_copy(u + slen + 1,t);
  86.  if (!env_add(u)) { alloc_free(u); return 0; }
  87.  return 1;
  88. }
  89.  
  90. int env_init()
  91. {
  92.  char **newenviron;
  93.  int i;
  94.  for (en = 0;environ[en];++en) ;
  95.  ea = en + 10;
  96.  newenviron = (char **) alloc((ea + 1) * sizeof(char *));
  97.  if (!newenviron) return 0;
  98.  for (en = 0;environ[en];++en)
  99.   {
  100.    newenviron[en] = alloc(str_len(environ[en]) + 1);
  101.    if (!newenviron[en])
  102.     {
  103.      for (i = 0;i < en;++i) alloc_free(newenviron[i]);
  104.      alloc_free(newenviron);
  105.      return 0;
  106.     }
  107.    str_copy(newenviron[en],environ[en]);
  108.   }
  109.  newenviron[en] = 0;
  110.  environ = newenviron;
  111.  env_isinit = 1;
  112.  return 1;
  113. }
  114.